package simple.number;

/**
 * <a href="https://leetcode.cn/problems/prime-number-of-set-bits-in-binary-representation/">762. 二进制表示中质数个计算置位</a>
 * 给你两个整数 left 和 right ，在闭区间 [left, right] 范围内，统计并返回 计算置位位数为质数 的整数个数。
 * 计算置位位数 就是二进制表示中 1 的个数。
 * 例如， 21 的二进制表示 10101 有 3 个计算置位。
 * 示例 1：
 *   输入：left = 6, right = 10
 *   输出：4
 *   解释：
 *     6 -> 110 (2 个计算置位，2 是质数)
 *     7 -> 111 (3 个计算置位，3 是质数)
 *     9 -> 1001 (2 个计算置位，2 是质数)
 *     10-> 1010 (2 个计算置位，2 是质数)
 *     共计 4 个计算置位为质数的数字。
 * 示例 2：
 *   输入：left = 10, right = 15
 *   输出：5
 *   解释：
 *     10 -> 1010 (2 个计算置位, 2 是质数)
 *     11 -> 1011 (3 个计算置位, 3 是质数)
 *     12 -> 1100 (2 个计算置位, 2 是质数)
 *     13 -> 1101 (3 个计算置位, 3 是质数)
 *     14 -> 1110 (3 个计算置位, 3 是质数)
 *     15 -> 1111 (4 个计算置位, 4 不是质数)
 *     共计 5 个计算置位为质数的数字。
 * 提示：
 *   1 <= left <= right <= 10^6
 *   0 <= right - left <= 10^4
 * @author 刘学松
 * @date 2023-06-27 14:33
 */
public class 二进制表示中质数个计算置位 {
    public int countPrimeSetBits(int left, int right) {
        boolean[] isZs = new boolean[20];
        isZs[2] = isZs[3] = isZs[5] = isZs[7] = isZs[11] = isZs[13] = isZs[17] = isZs[19] = true;
        int count = 0;
        // 保证左区间从偶数开始
        if ((left & 1) == 1) {
            if (isZs[Integer.bitCount(left)]) {
                count = 1;
            }
            ++left;
        }
        // 保证右区间是奇数
        if ((right & 1) == 0) {
            if (isZs[Integer.bitCount(right)]) {
                ++count;
            }
            --right;
        }
        while (left <= right) {
            // 统计二进制中1的个数，left为偶数
            int bitCount = Integer.bitCount(left);
            if (isZs[bitCount]) {
                ++count;
            }
            // 偶数+1，其实就是二进制1的个数+1，所以这里不用再统计一遍
            if (isZs[bitCount + 1]) {
                ++count;
            }
            // 一次统计了两个数，+2
            left += 2;
        }
        return count;
    }
}
